TypeScript type inference with lookup types and switch case